def forward(self, x):
        __breakpoint__ = paddle.shape(x)

        feats, shape = self.backbone(x)
        logits = self.head(feats[-1], shape[2:])
        
        logit_list = [
            F.interpolate(
                logit, __breakpoint__[2:], mode='bilinear') for logit in logits
        ]

        return logit_list

import sot
sot.add_breakpoint(__file__, 58)